"""
LeetCode 热题

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
"""
class Solution(object):
    """
    解决矩阵中只要有0就将整个行和列置为0的问题。

    方法:
        使用第一行和第一列来记录每行和每列是否应该被置为0。
    """

    def setZeroes(self, matrix):
        """
        将矩阵中出现0的行和列全部设置为0。

        参数:
            matrix: 二维整数数组，表示输入的矩阵。
        """
        # 检查矩阵及其首行是否为空
        if not matrix or not matrix[0]:
            return

        # 获取矩阵的行数和列数
        m, n = len(matrix), len(matrix[0])

        # 判断首行和首列是否包含0
        first_row_has_zero = any(matrix[0][j] == 0 for j in range(n))
        first_col_has_zero = any(matrix[i][0] == 0 for i in range(m))

        # 使用首行和首列记录哪些行和列需要置零
        for i in range(1, m):
            for j in range(1, n):
                if matrix[i][j] == 0:
                    matrix[i][0] = 0
                    matrix[0][j] = 0

        # 根据首行和首列的记录，将相关行和列置零
        for i in range(1, m):
            for j in range(1, n):
                if matrix[i][0] == 0 or matrix[0][j] == 0:
                    matrix[i][j] = 0

        # 如果首行原本包含0，将首行全部置零
        if first_row_has_zero:
            for j in range(n):
                matrix[0][j] = 0

        # 如果首列原本包含0，将首列全部置零
        if first_col_has_zero:
            for i in range(m):
                matrix[i][0] = 0


if __name__ == '__main__':
    a = Solution()
    test_matrix = [[1,1,1],[1,0,1],[1,1,1]]
    print("Original matrix:", test_matrix)
    a.setZeroes(test_matrix)
    print("Modified matrix:", test_matrix)
